package top.meeting.room.common.sql;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.Prop;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.druid.DruidPlugin;

import javax.sql.DataSource;

/**
 * 我有故事，你有酒么？
 * JKhaled created by yunqisong@foxmail.com 2017/7/24
 * FOR : 模型构造器入口内
 */
public class GeneratorMain {
    /**
     * 连接池
     *
     * @return
     */
    public static DruidPlugin druidPlugin() {
        Prop prop = PropKit.use("dev.db.properties");
        String jdbcUrl = "jdbc:mysql://" + prop.get("db.host") + ":" + prop.get("db.port") + "/"
                + prop.get("db.repertory") + "?characterEncoding=utf8&zeroDateTimeBehavior=convertToNull";
        return new DruidPlugin(jdbcUrl, prop.get("db.username"), prop.get("db.password"));
    }

    /**
     * 数据源
     *
     * @return
     */
    public static DataSource getDataSource() {
        DruidPlugin druidPlugin = druidPlugin();
        druidPlugin.start();
        return druidPlugin.getDataSource();
    }

    public static void main(String args[]) {
        // base model 所使用的包名

        String baseModelPackageName = "top.meeting.room.baseModel";
        // base model 文件保存路径

        String baseModelOutputDir = PathKit.getWebRootPath() + "/src/main/java/top/meeting/room/baseModel";

        // model 所使用的包名 (MappingKit 默认使用的包名)

        String modelPackageName = "top.meeting.room.common.model";
        // model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)

        String modelOutputDir = PathKit.getWebRootPath() + "/src/main/java/top/meeting/room/baseModel";

        // 创建生成器

        Generator generator = new Generator(getDataSource(), new JKBaseModelGenerator(baseModelPackageName, baseModelOutputDir));
        generator.setGenerateChainSetter(true);
        // 添加不需要生成的表名

//        generator.addExcludedTable("t_test");
        // 设置是否在 Model 中生成 dao 对象

        generator.setGenerateDaoInModel(true);
        // 设置是否生成链式 setter 方法

        generator.setGenerateChainSetter(true);
        // 设置是否生成字典文件

        generator.setGenerateDataDictionary(true);
        // 设置生成字典文件的目录

        generator.setDataDictionaryFileName("数据库表结构.txt");
        // 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser

        generator.setRemovedTableNamePrefixes("t_");
        // 生成

        generator.generate();
    }

}
